import requests
import json

def ask_ollama_with_file(file_path, question):
    """
    向 Ollama 发送文件内容和问题，获取 AI 回复
    
    Args:
        file_path (str): 文件路径
        question (str): 要询问的问题
    
    Returns:
        dict: Ollama 的响应结果
    """
    # 读取文件内容
    with open(file_path, 'r', encoding='utf-8') as f:
        file_content = f.read()
    
    # 构建提示词
    prompt = f"""你是一个专业的文档整理助手。请将提供的文档内容重新整理成结构清晰的markdown格式。

    整理要求：
    1. 保留原始页眉信息
    2. 按逻辑层次组织内容
    3. 使用规范的markdown标题格式
    4. 表格内容用markdown表格格式重新排版

    请整理以下内容：
    {file_content}"""
    
    # 发送请求
    response = requests.post(
        'http://localhost:11434/api/generate',
        json={
            'model': 'qwen:0.5b',
            'prompt': prompt,
            'stream': False
        }
    )
    
    return response.json()

def get_ollama_model_list():
    response = requests.get(
        'http://localhost:11434/api/tags'
    )
    return response.json()